// 有一幅以二维整数数组表示的图画，每一个整数表示该图画的像素值大小，数值在 0 到 65535 之间。

// 给你一个坐标 (sr, sc) 表示图像渲染开始的像素值（行 ，列）和一个新的颜色值 newColor，让你重新上色这幅图像。

// 为了完成上色工作，从初始坐标开始，记录初始坐标的上下左右四个方向上像素值与初始坐标相同的相连像素点，接着再记录这四个方向上符合条件的像素点与他们对应四个方向上像素值与初始坐标相同的相连像素点，……，重复该过程。将所有有记录的像素点的颜色值改为新的颜色值。

// 最后返回经过上色渲染后的图像。

// 示例 1:

// 输入: 
// image = [[1,1,1],[1,1,0],[1,0,1]]
// sr = 1, sc = 1, newColor = 2
// 输出: [[2,2,2],[2,2,0],[2,0,1]]
// 解析: 
// 在图像的正中间，(坐标(sr,sc)=(1,1)),
// 在路径上所有符合条件的像素点的颜色都被更改成2。
// 注意，右下角的像素没有更改为2，
// 因为它不是在上下左右四个方向上与初始点相连的像素点。
// 注意:

// image 和 image[0] 的长度在范围 [1, 50] 内。
// 给出的初始点将满足 0 <= sr < image.length 和 0 <= sc < image[0].length。
// image[i][j] 和 newColor 表示的颜色值在范围 [0, 65535]内。

#include <vector>
#include <queue>

using namespace std;

// BFS，超出时间限制
class Solution1 {
public:
    vector<vector<int>> floodFill(vector<vector<int>>& image, int sr, int sc, int newColor) {
        int m = image.size();
        int n = image[0].size();
        int oldColor = image[sr][sc];
        vector<vector<int>> res = image;
        vector<vector<int>> dirs{{0,-1}, {-1,0}, {0,1}, {1,0}};
        queue<pair<int, int>> q{{{sr, sc}}};
        while (!q.empty()) {
            int len = q.size();
            for (int i{0}; i < len; ++i) {
                auto t = q.front();
                q.pop();
                res[t.first][t.second] = newColor;
                for (auto dir : dirs) {
                    int x = t.first + dir[0];
                    int y = t.second + dir[1];
                    if (x < 0 || x >= m || y < 0 || y >= n || res[x][y] != oldColor) continue;
                    q.push({x,y});
                }
            }
        }
        return res;
    }
};

// DFS
class Solution {
public:
    vector<vector<int>> floodFill(vector<vector<int>>& image, int sr, int sc, int newColor) {
        if (image[sr][sc] == newColor) return image;
        func(image, sr, sc, image[sr][sc], newColor);
        return image;
    }
    void func(vector<vector<int>>& image, int i, int j, int oldColor, int newColor) {
        int m = image.size();
        int n = image[0].size();
        if (i < 0 || i >= m || j < 0 || j >= n || image[i][j] != oldColor) return;
        image[i][j] = newColor;
        func(image, i+1, j, oldColor, newColor);
        func(image, i-1, j, oldColor, newColor);
        func(image, i, j+1, oldColor, newColor);
        func(image, i, j-1, oldColor, newColor);
    }
};

class Solution {
public:
    vector<vector<int>> floodFill(vector<vector<int>>& image, int sr, int sc, int newColor) {
        if (image.empty() || image[0].empty()) return {};
        int m = image.size();
        int n = image[0].size();
        if (sr < 0 || sr >= m || sc < 0 || sc >= n || image[sr][sc] == newColor) {
            return image;
        }
        int oldColor = image[sr][sc];
        floodFill(image, sr, sc, m, n, oldColor, newColor);
        return image;
    }
    void floodFill(vector<vector<int>>& image, int sr, int sc, const int& m, const int& n, const int& oldColor, const int& newColor) {
        if (sr < 0 || sr >= m || sc < 0 || sc >= n || image[sr][sc] != oldColor) {
            return;
        }
        image[sr][sc] = newColor;
        floodFill(image, sr + 1, sc, m, n, oldColor, newColor);
        floodFill(image, sr - 1, sc, m, n, oldColor, newColor);
        floodFill(image, sr, sc + 1, m, n, oldColor, newColor);
        floodFill(image, sr, sc - 1, m, n, oldColor, newColor);
    }
};

class Solution {
public:
    vector<vector<int>> floodFill(vector<vector<int>>& image, int sr, int sc, int newColor) {
        int m = image.size();
        int n = image[0].size();
        int oldColor = image[sr][sc];
        if (oldColor != newColor) {
            floodFill(image, sr, sc, m, n, oldColor, newColor);
        }
        return image;
    }
    void floodFill(vector<vector<int>>& image, int sr, int sc, const int& m, const int& n, const int& oldColor, const int& newColor) {
        if (sr < 0 || sr >= m || sc < 0 || sc >= n || image[sr][sc] != oldColor) {
            return;
        }
        image[sr][sc] = newColor;
        floodFill(image, sr + 1, sc, m, n, oldColor, newColor);
        floodFill(image, sr - 1, sc, m, n, oldColor, newColor);
        floodFill(image, sr, sc + 1, m, n, oldColor, newColor);
        floodFill(image, sr, sc - 1, m, n, oldColor, newColor);
    }
};

class Solution {
public:
    vector<vector<int>> floodFill(vector<vector<int>>& image, int sr, int sc, int newColor) {
        int m = image.size();
        int n = image[0].size();
        int oldColor = image[sr][sc];
        if (oldColor == newColor) return image;
        queue<pair<int, int>> q{};
        q.push({sr, sc});
        image[sr][sc] = newColor;
        while (!q.empty()) {
            auto cur = q.front();
            q.pop();
            for (const auto& dir : dirs) {
                int new_sr = cur.first + dir[0];
                int new_sc = cur.second + dir[1];
                if (new_sr >= 0 && new_sr < m && new_sc >= 0 && new_sc < n && image[new_sr][new_sc] == oldColor) {
                    image[new_sr][new_sc] = newColor;
                    q.push({new_sr, new_sc});
                }
            }
        }
        return image;
    }
private:
    int dirs[4][2] = {
        {0, 1}, {0, -1}, {1, 0}, {-1, 0}
    };
};